MASTER_PORT=$(shuf -n 1 -i 10000-65535)
deepspeed --include='localhost:0,1,2,3,4,5,6,7' --master_port $MASTER_PORT /workspace/wavit/deepspeed_train_vq.py \
    --deepspeed_config=/workspace/wavit/config/deepspeed_fp16_zero2.json